# 文件夹下仅有文件的删除

# 问题：如果一个文件夹下仅有文件，自定义一个函数，用于删除整个文件夹

# 思路：获取文件夹中所有的文件，遍历删除，之后删除空的文件夹

import os


def my_rmdir(dir):
    files = os.listdir(dir)
    os.chdir(dir)
    # 删除文件夹中所有的文件
    for file in files:
        os.remove(file)
        print("删除成功！", file)
    # 删除空的文件夹
    os.chdir("..")
    os.rmdir(dir)


my_rmdir("E:\Desktop")
